TensorFlow Custom Model Builder: TensorFlow Custom Model Builder
Description
This function is used when a custom model is written.
Any custom model will be passed automatically into this function.
Usage
TensorFlowEstimator(n_classes, ...)
Arguments
n_classes
The number of classes for the target variable
...
Additional parameters including batch_size, steps, learning_rate, etc. A list can be found here: https://github.com/google/skflow/blob/master/skflow/__init__.py#L29